package com.ddd.mill.infrastructure.doc;

import com.fasterxml.jackson.databind.introspect.AnnotatedField;
import io.swagger.annotations.ApiModelProperty;
import org.springframework.stereotype.Component;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spi.schema.ModelPropertyBuilderPlugin;
import springfox.documentation.spi.schema.contexts.ModelPropertyContext;

import java.lang.reflect.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * swagger枚举处理
 * @author: zhoukunhua
 * @date: 2020-11-20 5:01 PM
 */
@Component
public class SwaggerDisplayConfig implements ModelPropertyBuilderPlugin {
    @Override
    public void apply(ModelPropertyContext context) {
        AnnotatedField field = context.getBeanPropertyDefinition().get().getField();
        Class<?> rawType = field.getRawType();
        if (!rawType.isEnum()) {
            return;
        }
        ApiModelProperty annotation = field.getAnnotation(ApiModelProperty.class);
        // Annotation是通过代理来设置并存储属性值的
        InvocationHandler ih = Proxy.getInvocationHandler(annotation);
        try {
            // Annotation所有属性的值都存储在InvocationHandler的memberValues属性，该属性类型为Map
            Field memberValuesField = ih.getClass().getDeclaredField("memberValues");
            memberValuesField.setAccessible(true);
            Map memberValues = (Map)memberValuesField.get(ih);
            String className = rawType.getDeclaredField("value").getType().getName();
            memberValues.put("dataType", className);
            Object[] enumConstants = rawType.getEnumConstants();
            Method getValue = rawType.getMethod("getValue");
            Method getDesc = rawType.getMethod("getDesc");
            List<String> descriptions = new ArrayList<>();
            List<String> allowableValues = new ArrayList<>();
            for (Object e : enumConstants){
                String v = getValue.invoke(e).toString();
                allowableValues.add(v);
                descriptions.add(v + "：" + getDesc.invoke(e));
            }
            memberValues.put("value", ((String)memberValues.get("value")).concat("。").concat(String.join("，", descriptions)));
            memberValues.put("allowableValues", String.join(",", allowableValues));
        } catch (NoSuchFieldException | IllegalAccessException | NoSuchMethodException | InvocationTargetException e) {
            e.printStackTrace();
        }
    }

    @Override
    public boolean supports(DocumentationType documentationType) {
        return true;
    }
}
